#!/bin/bash

# dataset_name=('semval-RE')
# dataset_name=('conll04')
# aug_rules=('add_triple' 'add_same_type_triple' 'remove_triple' 'replace_same_type_entity' 'swap_triple')
# timestamp=$(date +%s)
# echo $@
# echo $#

dataset_name=($1)
timestamp=$2
seed=$3
is_ft_generator=$4
aug_timies=$5
few_shot_num=$6

shift 6

aug_rules=$@
# echo $aug_rules

# for dataset_name in "${dataset_name[@]}"; do
#     python ./data_augmentation/generate_data.py \
#         --path_to_model_config="./configs/gen_data_config.yaml" \
#         --dataset_name=$dataset_name \
#         --rules="${aug_rules[@]}"
# done

for dataset in "${dataset_name[@]}"; do
    python ./data_augmentation/generate_data.py \
        --path_to_model_config="./configs/gen_data_config.yaml" \
        --dataset_name="$dataset" \
        --save_dir=$timestamp \
        --seed=$seed \
        --is_ft=$is_ft_generator \
        --aug_times=$aug_timies \
        --few_shot_num=$few_shot_num \
        --rules="$aug_rules"

done
